# 导入pandas
from pprint import pprint

import pandas as pd


# 读取excel文件 pd.read_excel(文件路径, sheet)
data = pd.read_excel('data/testcase.xlsx')
# data.loc[行标签,列标签]
# print(data.loc[0])  # 读取第一行信息
# print(data.loc[0, '用例编号'])  # 读取第一行中的用例编号

# data.loc[[行标签, 行标签], [列标签, 列标签]]
# print(data.loc[[0, 1, 3, 5]])
# print(data.loc[[0, 1, 3, 5], ['用例编号', '预期结果']])

# data.iloc[行索引, 列索引]
# print(data.iloc[1])  # 读取第二行信息
# print(data.iloc[1, 0])  # 读取第二行的用例编号

# data.iloc[[行索引1, 行索引2], [列索引1, 列索引2]]
# print(data.iloc[[1, 2, 3], [1, 2, 3, 4]])

# 将读取出的数据转化成列表字典
list1 = []
# 获取所有索引
index = data.index.values
# 通过索引读取信息
for i in index:
    # list1.append(data.loc[i].to_dict())  # data.loc[i].to_dict()转化成字典
    # list1.append(data.iloc[i].tolist())   # data.iloc[i].tolist()转化成列表列表
    list1.append(data.iloc[i].values.tolist())   # data.iloc[i].values.tolist()转化成列表列表
pprint(list1)

# 列表推导式
# list2 = [data.loc[i].to_dict() for i in data.index.values]
# print(list2)